package com.sdy.dcsb.web.util;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.sdy.common.utils.StringUtil;
import org.jdom.Attribute;
import org.jdom.Document;
import org.jdom.Element;
import org.jdom.JDOMException;
import org.jdom.input.SAXBuilder;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;

/**
 * json 相关工具类
 * @author Snaky
 * @date 2020/4/17 10:23
 */
public class JsonUtil {

    /**
     * jsonArray 合并
     */
    public static JSONArray mergeArray(JSONArray array1, JSONArray array2){
        if(array1 == null || array2 == null){
            return null;
        }
        JSONArray result = JSONArray.parseArray(array2.toString());
        if(result.size() == 0){
            return array1;
        }
        for (int i = 0; i < array1.size(); i++) {
            JSONObject arrayI = array1.getJSONObject(i);
            for (int j = 0; j < array2.size(); j++) {
                JSONObject oldJson = array2.getJSONObject(j);
                JSONObject tempJson = JSONArray.parseObject(oldJson.toString());
                tempJson.putAll(arrayI);
                result.remove(oldJson);
                result.add(tempJson);
            }
        }
        return result;
    }

    /**
     * 数组去重
     * @param array 源数组
     * @return 去重后数组
     */
    public static JSONArray distinctJSONArray(JSONArray array){
        if(array == null || array.size() <= 1){
            return array;
        }
        return JSONArray.parseArray(JSON.toJSONString(new HashSet<>(array)));
    }

    public static String xml2JSON(String xml) {
        if(StringUtil.isBlank(xml)){
            return xml;
        }
        JSONObject json = new JSONObject();
        try {
            byte[] byteStream = xml.getBytes("utf-8");
            InputStream is = new ByteArrayInputStream(byteStream);
            SAXBuilder sb = new SAXBuilder();
            Document doc = sb.build(is);
            Element root = doc.getRootElement();
            json.put(root.getName(), iterateElement(root));
            return json.toJSONString();
        } catch (Exception e1) {
            e1.printStackTrace();
        }
        return xml;
    }

    @SuppressWarnings("unchecked")
    private static JSONObject iterateElement(Element element) {
        List<Element> node = element.getChildren();
        Element et = null;
        JSONObject obj = new JSONObject();
        //其内的值可能为一个元素，也可能为一个list,故不设泛型
        List list = null;
        //遍历当前元素
        for (int i = 0; i < node.size(); i++) {
            list = new LinkedList<>();
            et = node.get(i);

            //获取元素内的属性
            if (et.getAttributes().size() > 0) {
                //获取当前元素的所有的属性值
                List<Attribute> attributes = et.getAttributes();
                //每一个属性都是一个jsonobject
                JSONObject attrvalue = new JSONObject();
                //属性值最终要合成的list
                List<JSONObject> attrvalues = new ArrayList<>();
                for (Attribute attr : attributes) {
                    //将参数的名和值封装为一个jsonobject
                    attrvalue.put(attr.getName(),attr.getValue());
                }
                attrvalues.add(attrvalue);
                obj.put(et.getName(), attrvalues);
            }
            //判断元素是否为空
            if (StringUtil.isBlank(et.getTextTrim())) {
                if (et.getChildren().size() == 0) {
                    continue;
                }
                //判断obj内是否有与当前元素重名的元素
                if (obj.containsKey(et.getName())) {
                    //若存在，就在之前的list后拓展
                    list = (List<JSONObject>) obj.get(et.getName());
                }
                list.add(iterateElement(et));
                obj.put(et.getName(), list);
            } else {
                if (obj.containsKey(et.getName())) {
                    list = (List<JSONObject>) obj.get(et.getName());
                }
                list.add(et.getTextTrim());
                obj.put(et.getName(), list);
            }
        }
        return obj;
    }

}
